#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <unistd.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <pthread.h>
#include <stdbool.h>

int num = 0;
int p;
int sockfd;//返回的套接字
struct sockaddr_in Rose_addr ;//Rose的地址
struct sockaddr_in from_addr[10] ;//用来存放Rose的IP以及port
int size = sizeof(struct sockaddr_in );//方便后面操作

//线程回调函数
void * func (void * arg)
{
    char * Jack_send_msg = calloc(128,1);
    int Jack_send_byte = -1;


    while(1)
    {
        printf("请输入#查看在线好友\n");
        while(getchar()!= '#'); //获取不到就什么也不干
        printf("************在线列表*******************\n");
        for(int i = 0;i < num;i++)
            {
                printf("from_addr[%d]:port=%d\n",i,ntohs(from_addr[i].sin_port)); //将存取信息与Rose进行比较，看看出错没有
            }
        printf("**************************************\n");

        printf("请选择你要给谁发送信息\n");
        scanf("%d",&p);
        while(getchar() != '\n');

        printf("Jacy 请输入要发送的信息:\n");
        fgets(Jack_send_msg,128,stdin);
        
        Jack_send_byte = sendto(sockfd, Jack_send_msg, strlen(Jack_send_msg) , 0 , (struct sockaddr *)&from_addr[p], size);
        if(Jack_send_byte == -1)
        {
            perror("Jcay recvive fail\n");
            continue;
        }
        else
        {
            printf("Jacy成功发送信息,共%d个字节\n",Jack_send_byte);
        }
    
    }

    return 0;

}
//判断并存放函数
bool Judge (struct sockaddr_in addr)
{
    printf("开始判断\n");
   
    for(int i=0;i<9;i++)//判断信息是否为同一个客户端发出，如果是可以避免重复存放客户端信息
        {
            if(memcmp((void *)&from_addr[i], (void *)&addr, sizeof(addr))) //如果为非0值则表示两者不相同
            {
                //from_addr[i] = Rose_addr;
                continue;  //继续轮换挨个儿比较是否相同
            }
            else
            {
                printf("不可\n");
                return true ;//如果相同那接下来也没有比较的必要了,返回真值
            } 
            
            printf("可存放\n");
            return false; //不相同就可以存放，返回false值；

        }
 
}


int main(int argc, char const *argv[])
{
    //建立网络接口（油箱）
    sockfd = socket(AF_INET, SOCK_DGRAM , 0);//返回待连接套接字
    if(sockfd == -1)
    {
        perror("Jack socket fail\n");
        return -1;
    }

    //设置IP地址以及端口号
    // struct sockaddr_in {
    // sa_family_t     sin_family;     /* AF_INET */
    // in_port_t       sin_port;       /* Port number */
    // struct in_addr  sin_addr;       /* IPv4 address */
    // };

    struct sockaddr_in my_addr = {0};
    my_addr.sin_family = AF_INET;  //网络簇 IPV4协议
    my_addr.sin_port = htons(65000);//设置网络端口号，并且转化网络字节序二进制
    my_addr.sin_addr.s_addr = htonl(INADDR_ANY);//当前服务端下所有IP地址均能够接受信息


    //绑定油箱和地址
    int retval  = bind(sockfd, (struct sockaddr *)&my_addr,size);
    if(retval)
    {
        perror("bind fail\n");
        return -1;
    }

    //创建线程进行发送信息
    pthread_t Tid;
    int ret = pthread_create(&Tid,NULL,func,NULL);
    if(ret)
    {
        fprintf(stderr,"pthread_create error:%s",strerror(ret)); //打印出错误原因
        close(sockfd);
        return -1;
    }


    //坐等接受信息
    char * rev_msg = calloc(128,1);
    
    while(1)
    {
       
        //printf("Jacy等待消息进来\n");
        int rec_byte = recvfrom(sockfd, rev_msg, 128 , 0 , (struct sockaddr *)&Rose_addr,&size);
        if(rec_byte == -1)
        {
            perror("recvive fail\n");
            return 0;
        }
        else
        {
            printf("recive sucess\n");
            printf("Rose_IP = %s\nport=%d\nrec_msg:%s ",inet_ntoa(Rose_addr.sin_addr),ntohs(Rose_addr.sin_port),rev_msg);
            if(!Judge(Rose_addr) && num < 9)
            {
                printf("存放成功\n");
                from_addr[num++] = Rose_addr;
            }

            for(int i = 0;i < num;i++)
            {
                printf("from_addr[%d]:port=%d\n",i,ntohs(from_addr[i].sin_port)); //将存取信息与Rose进行比较，看看出错没有
            }
        }

    }
    //关闭套接字文件
    close(sockfd);

    return 0;
}

